from base.base_page import Base
import page, time, allure

class Add_User_Page(Base):
    def __init__(self,driver):
        Base.__init__(self,driver)

    def click_add(self):
        # 点击添加用户按钮
        self.click_element(page.add_user_button)

    def click_save_local(self):
        # 点击本地保存
        self.click_element(page.local_button)
        self.driver.keyevent(4)
        # # 判断是否在用户详情页,是则说明返回不成功，再次执行返回
        # if self.if_disp(page.usr_edit_btn):
        #     allure.attach("描述", "点击手机返回按钮")
        #     self.driver.keyevent(4)

    def get_user_list(self):
        # 获取联系人列表
        user_data = self.find_elements_o(page.user_text)
        return [i.text for i in user_data]

    def input_user_info(self, name, phone):
        # 输入用户名
        allure.attach("描述", "输入用户名")
        self.input_text(page.send_name, name)
        # 输入手机号
        allure.attach("描述", "输入手机号")
        self.input_text(page.send_phone, phone)
        # 点击返回保存
        allure.attach("描述", "保存，并点击返回按钮")
        # self.click_element(page.click_save_back)
        self.click_element(page.local_button)
        self.driver.keyevent(4)
